set(TARGET yolo_v8_trt)

set(HDRs 
    decode_yolov8.h
    yolov8.h
   )

set(SRCs 
   decode_yolov8.cu
   yolov8.cpp 
   )

add_library(${TARGET} ${SRCs} ${TRT_SAMPLES_CPPS})

target_include_directories(${TARGET} PRIVATE ${HDRs}  PRIVATE ${ALL_INC_DIRS})
target_link_libraries(${TARGET} ${ALL_LIBS} ${OpenCV_LIBRARIES})
set_property(TARGET ${TARGET} PROPERTY CUDA_ARCHITECTURES 50 61 72 75)
target_compile_options(${TARGET} PUBLIC 
   $<$<COMPILE_LANGUAGE:CUDA>:--default-stream per-thread -lineinfo; --use_fast_math --disable-warnings>)